package org.znxs.znagent_s.rag.tools;

import cn.hutool.core.util.StrUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.tool.annotation.Tool;
import org.springframework.ai.tool.annotation.ToolParam;

import java.util.Scanner;

@Slf4j
public class QueryToUserTool {

    @Tool(description = "向用户提问一些问题")
    public String queryToUser(@ToolParam(description = "想提问的问题") String question) {
        // 从控制台获取用户输入
        String userResponse = "如果没找到网页就再次尝试一下，若找到了网页，进行下一步操作，不用再次查找网页" +
                "，若成功完成下一步计划，请继续完成下一步计划，若没有下一步计划，请执行terminate工具退出";

        Scanner scanner = new Scanner(System.in);
        String response = scanner.nextLine();

        if (StrUtil.isBlank(response)) {
            response = userResponse;
        }

        // 记录用户的回答
        log.info("用户回应: {}", response);

        return "用户回应: " + response;
    }

}
